package cn.arnold.stack;

import java.util.Scanner;

public class StackTest {

    public static void main(String[] args) {
        //测试一下Stack 是否正确
        //先创建一个Stack对象->表示栈
        Stack stack = new Stack(4);
        int key = 0;
        boolean loop = true; //控制是否退出菜单
        Scanner scanner = new Scanner(System.in);

        while (loop) {
            System.out.println("1: 表示显示栈");
            System.out.println("2: 退出程序");
            System.out.println("3: 表示添加数据到栈(入栈)");
            System.out.println("4: 表示从栈取出数据(出栈)");
            System.out.println("请输入你的选择");
            key = scanner.nextInt();
            switch (key) {
                case 1:
                    stack.showStack();
                    break;
                case 3:
                    System.out.println("请输入一个数");
                    int value = scanner.nextInt();
                    stack.push(value);
                    break;
                case 4:
                    try {
                        int res = stack.pop();
                        System.out.printf("出栈的数据是 %d\n", res);
                    } catch (Exception e) {
                        // TODO: handle exception
                        System.out.println(e.getMessage());
                    }
                    break;
                case 2:
                    scanner.close();
                    loop = false;
                    break;
                default:
                    break;
            }
        }

        System.out.println("程序退出~~~");
    }

}
